package chapter02

object Test05_Calculator {
  def main(args: Array[String]): Unit = {
    def inputAndOutputAndCalcu(f:(Int,Int)=>Int): Any = {
      println("请输入第一个数字：")
      val a = scala.io.StdIn.readInt()
      println("请输入第二个数字：")
      val b = scala.io.StdIn.readInt()
      if(b!=0){
        f(a,b)
      }else{
        println("除数不能为0")
      }
    }
    var flag = true
    while(flag){
      print(
        """欢迎来到计算器程序
          |1、进行加法操作
          |2、进行减法操作
          |3、进行乘法操作
          |4、进行除法操作
          |5、退出程序
          |请输入操作码：
          |""".stripMargin)
      //接收操作码
      val code = scala.io.StdIn.readInt()
      if(code==1){
        println("结果为："+inputAndOutputAndCalcu(_+_))
      }else if(code==2){
        println("结果为："+inputAndOutputAndCalcu(_-_))
      }else if(code==3){
        println("结果为："+inputAndOutputAndCalcu(_*_))
      }else if(code==4){
        println("结果为："+inputAndOutputAndCalcu(_/_))
      }else if(code==5){
        println("退出程序")
        flag=false
      }else{
        println("输入的操作码不正确，请重新输入")
      }
    }
  }
}
